# import dataloader
import models
# import loss

class register:
    def __init__(self):
        self.all_dataset = ['lastfm', 'gowalla', 'yelp2018', 'amazon-book']
        # self.all_dataset = [dataloader.la]


    def check_model(self, model_name):
        if model_name not in self.all_models:
            raise NotImplementedError(f"Haven't supported {model_name} yet!, try {self.all_models.items()}")

    def check_dataset(self, dataset_name):
        if dataset_name not in self.all_dataset:
            raise NotImplementedError(f"Haven't supported {dataset_name} yet!, try {self.all_dataset}")


